Personal Computer World 2009 February/PCWFEB09.iso/Software/Freeware/Ultra Defrag 2/ultradefrag-2.0.0.bin.i386.exe/scripts/udreportcnv.lua

Lua script | 2008-11-04 | 6.7 KB | 261 lines

#!/usr/local/bin/lua
--[[
  udreportcnv.lua - UltraDefrag report converter.
  Converts lua reports to HTML and other formats.
  Copyright (c) 2008 by Dmitri Arkhangelski ([e-mail]).

  This program is free software; you can redistribute it and/or modify
  it under the terms of the GNU General Public License as published by
  the Free Software Foundation; either version 2 of the License, or
  (at your option) any later version.

  This program is distributed in the hope that it will be useful,
  but WITHOUT ANY WARRANTY; without even the implied warranty of
  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  GNU General Public License for more details.

  You should have received a copy of the GNU General Public License
  along with this program; if not, write to the Free Software
  Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
--]]

-- USAGE: lua udreportcnv.lua <luar file with full path> <path to Windows directory> [-v]

-- parse command line
assert(arg[1],"Lua Report file name must be specified!")
assert(arg[2],"Path to the Windows directory\nmust be specified as second parameter!")

-- read options
dofile(arg[2] .. "\\UltraDefrag\\options\\udreportopts.lua")

-- source file reading
dofile(arg[1])


-- common functions

function write_ansi(f, ...)
    for i, v in ipairs(arg) do f:write(v) end
end

function write_unicode(f, ...)
    local b, j
    for i, v in ipairs(arg) do
        j = 1
        while true do
            b = string.byte(v, j)
            if b == nil then break end
            f:write(string.char(b), "\0")
            j = j + 1
        end
    end
end

-- write filename to html table; maximum 50 characters per line
function write_ansi_name(f, name)
    local i, j, k, n, len
    local N = max_chars_per_line
    len = string.len(name)
    i = 1
    while len > 0 do
        if len <= N then f:write(string.sub(name,i,i + len - 1)) ; break end
        j = string.find(name,"\\",i,true)
        if j == nil or j > (i + N) then
            n = N
        else
            k = j
            while true do
                k = string.find(name,"\\",k + 1,true)
                if k == nil then break end
                if k <= (i + N) then j = k else break end
            end
            n = j - i + 1
        end
        f:write(string.sub(name,i,i + n - 1), "<br />")
        if n == 0 then break end -- erroneous situation
        i = i + n
        len = len - n
    end
end

function write_chars(f, chars, i, j)
    local k
    for k=i,j do f:write(chars[k]) end
end

function find_char(chars, ch, i)
    local k = i
    local result = nil
    while chars[k] ~= nil do
        if chars[k] == ch and chars[k + 1] == '\0' then
            result = k ; break
        end
        k = k + 1
    end
    return result
end

function write_unicode_name(f, chars, len)
    local i, j, k, n
    local N = max_chars_per_line * 2
    i = 1
    while len > 0 do
        if len <= N then write_chars(f,chars,i,i + len - 1) ; break end
        j = find_char(chars,'\\',i)
        if j == nil or j > (i + N) then
            n = N
        else
            k = j
            while true do
                k = find_char(chars,'\\',k + 1)
                if k == nil then break end
                if k <= (i + N) then j = k else break end
            end
            n = j - i + 1 + 1 -- because unicode backslash is '\\\0'
        end
        write_chars(f,chars,i,i + n - 1)
        write_unicode(f,"<br />")
        if n == 0 then break end -- erroneous situation
        i = i + n
        len = len - n
    end
end

-- converters

table_head = [[
<tr>
<td class="c"><a href="javascript:sort_items('fragments')" style="color: #0000FF"># fragments</a></td>
<td class="c"><a href="javascript:sort_items('name')" style="color: #0000FF">filename</a></td>
<td class="c"><a href="javascript:sort_items('comment')" style="color: #0000FF">comment</a></td>
</tr>
]]

end_of_page = [[
</center>
<script type="text/javascript">
init_sorting_engine();
</script>
</body></html>
]]

links_x1 = [[
<table width="100%"><tbody>
<tr>
<td style="text-align: left"><a href="http://ultradefrag.sourceforge.net" style="color: #0000FF">Visit our Homepage</a></td>
<td style="text-align: center"><a
]]

links_x2 = [[
style="color: #0000FF">View report options</a></td>
<td style="text-align: right">
<a href="http://www.lua.org/" style="color: #0000FF">Powered by Lua</a>
</td>
</tr>
</tbody></table>
]]

function produce_html_output()
    local filename
    local pos = 0
    local js
    local links = links_x1 .. "href=\"file:///" .. arg[2]
    links = links .. "\\UltraDefrag\\options\\udreportopts.lua\" " .. links_x2

    repeat
        pos = string.find(arg[1],"\\",pos + 1,true)
        if pos == nil then filename = "fraglist.htm" ; break end
    until string.find(arg[1],"\\",pos + 1,true) == nil
    filename = string.sub(arg[1],1,pos) .. "fraglist.htm"

    -- note that 'b' flag is needed for utf-16 files
    local f = assert(io.open(filename,"wb"))

    local write_data
    if use_utf16 == 0 then
        write_data = write_ansi
    else
        write_data = write_unicode
    end

    if(enable_sorting == 1) then
        -- read udsorting.js file contents
        local f2 = assert(io.open(arg[2] .. "\\UltraDefrag\\scripts\\udsorting.js", "r"))
        js = f2:read("*all")
        f2:close()
    else
        js = "function init_sorting_engine(){}\nfunction sort_items(criteria){}\n"
    end
    write_data(f,
        "<html><head><title>Fragmented files on ", volume_letter,
        ":</title>\n", style,
        "<script language=\"javascript\">\n", js,
        "</script>\n</head>\n",
        "<body>\n<center>\n", title_tags.open,
        "Fragmented files on ", volume_letter,
        ":", title_tags.close,
        "\n", links, "\n",
        "<div id=\"for_msie\">\n",
        "<table id=\"main_table\" ", table_style, ">\n",
        table_head
        )
    for i, v in ipairs(files) do
        local class
        if v.filtered == 1 then class = "f" else class = "u" end
        write_data(f,
            "<tr class=\"", class, "\"><td class=\"c\">", v.fragments,
            "</td><td>"
            )
        if use_utf16 == 0 then
            -- each <> brackets must be replaced with square brackets
            local tmp = string.gsub(v.name,"<","[")
            tmp = string.gsub(tmp,">","]")
            if split_long_names == 1 then
                write_ansi_name(f,tmp)
            else
                f:write(tmp)
            end
        else
            if split_long_names == 1 then
                local chars = {}
                local i = 1
                for j, b in ipairs(v.uname) do
                    chars[i] = string.char(b)
                    i = i + 1
                end
                write_unicode_name(f,chars,i - 1)
            else
                for j, b in ipairs(v.uname) do
                    f:write(string.char(b))
                end
            end
        end
        write_data(f,
            "</td><td class=\"c\">", v.comment,
            "</td></tr>\n"
            )
    end
    write_data(f,
        --table_head,
        "</table>\n</div>\n", links, "\n"
        )
    write_data(f,end_of_page)
    f:close()

    -- if option -v is specified, open report in default web browser
    if arg[3] == "-v" then
        if os.shellexec ~= nil then
            os.shellexec(filename,"open")
        else
            os.execute("cmd.exe /C " .. filename)
        end
    end
end


-- main source code
-- convertion to other (readable) formats
if produce_html == 1 then
    produce_html_output()
end